import os
from dotenv import load_dotenv

load_dotenv()


class Config:
    # 阿里云百练API配置
    ALIBABA_CLOUD_API_KEY = os.getenv('DASHSCOPE_API_KEY')
    ALIBABA_CLOUD_ENDPOINT = "https://dashscope.aliyuncs.com/api/v1/services/aigc/text-generation/generation"

    # MongoDB配置
    MONGODB_URL = os.getenv('MONGODB_URL', 'mongodb://localhost:27017/')
    MONGODB_DATABASE = 'sentiment_analysis'

    # 数据采集配置
    CRAWL_INTERVAL = 3600  # 1小时采集一次
    MAX_ARTICLES_PER_SOURCE = 150
    RELAXED_COLLECTION = True  # 宽松采集：标题命中关键词即可放行

    # 新能源汽车相关关键词（修正乱码并扩充）
    NEW_ENERGY_KEYWORDS = [
        '新能源汽车', '电动车', '电动汽车', '特斯拉', '比亚迪', '蔚来', '理想', '小鹏',
        '新能源车', '纯电', '混动', '插电混动', '氢能源', '燃料电池',
        '充电桩', '电池技术', '续航里程', '自动驾驶', '智能汽车',
        '高压快充', '换电', '800V', '智驾', '三元锂', '固态电池'
    ]

    # 数据源配置（扩充新能源/汽车资讯频道）
    NEWS_SOURCES = [
        'https://www.autohome.com.cn/news/',
        'https://www.autohome.com.cn/ev/',
        'https://www.pcauto.com.cn/news/',
        'https://news.yiche.com/',
        'https://auto.sina.com.cn/news/',
        'https://auto.sohu.com/newenergy/',
        'https://www.gasgoo.com/news/',
        'https://www.163.com/auto/'
    ]

    # RSS 源（可按需添加）
    RSS_SOURCES = [
        # 示例：'https://www.caranddriver.com/rss/all.xml',
        # 示例：'https://www.autoblog.com/rss.xml'
    ]

    # 社交媒体配置
    WEIBO_KEYWORDS = ['新能源汽车', '电动车', '特斯拉', '比亚迪', '蔚来']

    # 分析配置
    SENTIMENT_THRESHOLD = 0.1  # 情感分析阈值
    TOPIC_NUM = 10  # 主题数量
    TREND_DAYS = 30  # 趋势分析天数
